#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# @Project : chat_model 
# @File    : schema.py
# @IDE     : PyCharm 
# @Author  :ZH
# @Time    : 2025/1/10 13:12
from datetime import datetime

from pydantic import BaseModel
from typing import Optional, List


class VectorStoreListRequests(BaseModel):
    page: int = 1
    limit: int = 10


class VectorStoreListResponseBase(BaseModel):
    id: Optional[int] = None
    update_date: Optional[datetime] = None
    vector_store_name: Optional[str] = None
    vector_store_type: Optional[str] = None
    vector_store_index: Optional[list] = None
    description: Optional[str] = None
    is_default: Optional[bool] = None


class VectorStoreListResponse(BaseModel):
    code: int
    message: str
    data: Optional[List[VectorStoreListResponseBase]] = None
